#!/usr/bin python3
# -*- coding:UTF-8 -*-
# Author: nigo
import pandas as pd

df = pd.DataFrame([[4, 9]] * 3, columns=['A', 'B'])
result = df.apply(lambda x: x * x + 1)
print(result)
